package cn.e4j.core;

import cn.e4j.kernel.generator.GenerateParams;
import com.baomidou.mybatisplus.annotation.DbType;
import com.baomidou.mybatisplus.annotation.FieldFill;
import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.generator.AutoGenerator;
import com.baomidou.mybatisplus.generator.config.*;
import com.baomidou.mybatisplus.generator.config.po.TableFill;
import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;

import java.util.ArrayList;

/**
 * 简单代码生成器，service不生成接口
 *
 * @author wenchuan.li
 * @Date 2018/11/20 11:18
 */
public class SimpleGenerator {

    public static void doGeneration(GenerateParams generateParams) {

        AutoGenerator mpg = new AutoGenerator();

        // 全局配置
        GlobalConfig gc = new GlobalConfig();
        gc.setOutputDir(generateParams.getOutputDirectory());
        gc.setFileOverride(true);
        gc.setActiveRecord(false);// 不需要ActiveRecord特性的请改为false
        gc.setBaseResultMap(true);// XML ResultMap
        gc.setBaseColumnList(true);// XML columList
        gc.setEnableCache(false);
        gc.setOpen(false);
        gc.setActiveRecord(true);
        gc.setAuthor(generateParams.getAuthor());
        gc.setIdType(IdType.INPUT);// ID idType

        // 自定义文件命名，注意 %s 会自动填充表实体属性！
        if (generateParams.getGeneratorInterface()) {
            gc.setServiceName("%sService");
            gc.setServiceImplName("%sServiceImpl");
        } else {
            gc.setServiceName("%sService");
            gc.setServiceImplName("%sService");
        }
        mpg.setGlobalConfig(gc);

        // 数据源配置
        DataSourceConfig dsc = new DataSourceConfig();
        // 设置成oracle时 一直走mysql驱动，把pom里面的依赖注释掉就好了
        dsc.setDbType(generateParams.getJdbType());

        dsc.setDriverName(generateParams.getJdbcDriver());
        dsc.setUrl(generateParams.getJdbcUrl());
        dsc.setUsername(generateParams.getJdbcUserName());
        dsc.setPassword(generateParams.getJdbcPassword());
        mpg.setDataSource(dsc);

        // 策略配置
        StrategyConfig strategy = new StrategyConfig();
        strategy.setCapitalMode(false);
        strategy.setEntityLombokModel(true);

        // 此处可以移除表前缀表前缀
        strategy.setTablePrefix(generateParams.getRemoveTablePrefix());

        // 表名生成策略 开启自动驼峰命名
        strategy.setNaming(NamingStrategy.underline_to_camel);
        strategy.setColumnNaming(NamingStrategy.nochange);//不设置默认表字段

        // 需要生成的表
        strategy.setInclude(generateParams.getIncludeTables());



        // 公共字段填充
        // bug bean里面的注解会出现丢失的情况 需要手动添加
        ArrayList<TableFill> tableFills = new ArrayList<>();
        tableFills.add(new TableFill("CREATE_TIME", FieldFill.INSERT));
        tableFills.add(new TableFill("ID", FieldFill.INSERT));

        tableFills.add(new TableFill("CREATE_BY", FieldFill.INSERT));
        tableFills.add(new TableFill("UPDATE_BY", FieldFill.INSERT_UPDATE));

        tableFills.add(new TableFill("CREATE_DATE", FieldFill.INSERT));
        tableFills.add(new TableFill("UPDATE_DATE", FieldFill.INSERT_UPDATE));

        strategy.setTableFillList(tableFills);

        mpg.setStrategy(strategy);

        // 自定义模板配置，可以 copy 源码 mybatis-plus/src/main/resources/templates 下面内容修改，
        // 放置自己项目的 src/main/resources/templates 目录下, 默认名称一下可以不配置，也可以自定义模板名称
        TemplateConfig tc = new TemplateConfig();
        tc.setController(null);

        if (!generateParams.getGeneratorInterface()) {
            tc.setService(null);
            tc.setServiceImpl("/templates/NoneInterfaceServiceImpl.java");
        }

        tc.setEntity("/templates/entity.java");

        //如上任何一个模块如果设置 空 OR Null 将不生成该模块。
        mpg.setTemplate(tc);

        // 包配置
        PackageConfig pc = new PackageConfig();
        pc.setParent(generateParams.getParentPackage());
        pc.setModuleName("");
        pc.setXml("mapper.mapping");
        pc.setEntity("entity");

        if (generateParams.getGeneratorInterface()) {
            pc.setServiceImpl("service.impl");
            pc.setService("service");
        } else {
            pc.setServiceImpl("service");
            pc.setService("service");
        }

        mpg.setPackageInfo(pc);

        // 执行生成
        mpg.execute();
    }

    public static void main(String[] args) {
        GenerateParams generateParams = new GenerateParams();
        generateParams.setAuthor("caijy");

//        generateParams.setJdbType(DbType.SQL_SERVER);
//        generateParams.setJdbcDriver("com.microsoft.sqlserver.jdbc.SQLServerDriver");
//        generateParams.setJdbcUrl("jdbc:sqlserver://172.16.100.195:1433;databaseName=E4J_CRM_MINGYI");
//        generateParams.setJdbcUserName("sa");
//        generateParams.setJdbcPassword("dcms00:");
        generateParams.setJdbType(DbType.ORACLE);
        generateParams.setJdbcDriver("oracle.jdbc.OracleDriver");
        generateParams.setJdbcUrl("jdbc:oracle:thin:@172.16.100.195:1521:MPL");
        generateParams.setJdbcUserName("e4jcrmbase");
        generateParams.setJdbcPassword("E4j_crm2019");
        generateParams.setParentPackage("cn.e4j.modular.erp");
        generateParams.setIncludeTables(new String[]{
                "GWBE_T"
        });

        // sql server 无法生成
        // 插件bug,com\baomidou\mybatisplus\generator\config\querys\SqlServerQuery.java
        // tableFieldsSql()中 拼接的sql存在 表名 字段名是大写的无法和数据库匹配，需要修改重新打包
        // 等待作者修复，若未修改就本地修改打包
//        generateParams.setJdbType(DbType.SQL_SERVER);
//        generateParams.setJdbcDriver("com.microsoft.sqlserver.jdbc.SQLServerDriver");
//        generateParams.setJdbcUrl("jdbc:sqlserver://172.16.100.195:1433;databaseName=IQS_6.0");
//        generateParams.setJdbcUserName("sa");
//        generateParams.setJdbcPassword("dcms00:");
//        generateParams.setParentPackage("cn.e4j.modular.iqs");
//        generateParams.setIncludeTables(new String[]{
//                "GWAA_T"
//        });

//        generateParams.setJdbType(DbType.SQL_SERVER);
//        generateParams.setJdbcDriver("com.microsoft.sqlserver.jdbc.SQLServerDriver");
//        generateParams.setJdbcUrl("jdbc:sqlserver://172.16.100.195:1433;databaseName=E4J_CRM_SANMU");
//        generateParams.setJdbcUserName("sa");
//        generateParams.setJdbcPassword("dcms00:");
//        generateParams.setParentPackage("cn.e4j.modular.crm");
//        generateParams.setIncludeTables(new String[]{
//                "GWDA_T"
//        });

        SimpleGenerator.doGeneration(generateParams);
    }

}